from pydantic import BaseModel, Field
from typing import List, Dict

class GetEmbeddingRequest(BaseModel):
    text: str

class LLMResponse(BaseModel):
    thinking: str = Field(description="详细说明您是如何根据上下文得出答案的思考过程。")
    final_answer: str = Field(description="回答问题")
    sources: str = Field(default="", description="信息来源")

class ChatRequest(BaseModel):
    question: str = Field(description="用户的问题")
    history: List[Dict[str, str]] = []


class ChatResponse(BaseModel):
    answer: str = Field(description="回答")
    thinking: str = Field(description="思考")
    sources: str = Field(default="", description="来源")


